// Copyright 2022 jeff.li. and/or its affiliates.

#pragma once

#include <tbir/pipeline/op_kernel.h>
#include <unordered_map>

namespace tbir::runtime {

    class DeviceOp : public OpKernel {
    public:
        void Init() override;

        RTValue Process(PyArgs inputs) const override;

    private:
        int device_id_;
    };
}  // namespace tbir::runtime